import CONSTANTESBATIMENTS
from Etoile import *
from Planete import *
from Joueur import *
from unite import *



class Structure(object):
     def __init__(self, type, etoile):
         self.type=type
         self.parent=etoile
         self.proprietaire =self.parent.proprietaire
         self.vie=CONSTANTESBATIMENTS.structures[type]['vie']
         self.bouclier=CONSTANTESBATIMENTS.structures[type]['bouclier']
         self.technologie=[]
         
     def ajouterTechnologie(self, technologie):
         self.technologie.append(technologie)         
   
class Technologie(object):
    def __init__(self,nom,gold,minerai,energie):
        self.nom=nom
        self.status=False
        self.gold=gold
        self.minerai=minerai
        self.energie=energie
      
class Batiment(object):
     def __init__(self, type, planete):
         self.etat="CONSTRUCTION"         
         self.tempsConstructionRestant=20
         self.etatAffichage="(construction) "+str(self.tempsConstructionRestant)
         self.type=type
         self.parent=planete
         self.vie=CONSTANTESBATIMENTS.batiments[type]['vie']
         self.bouclier=CONSTANTESBATIMENTS.batiments[type]['bouclier']
         self.vitesse=CONSTANTESBATIMENTS.batiments[type]['vitesse']
         if (self.type == 'tour'):
             self.attaque = CONSTANTESBATIMENTS.batiments[type]['attaque'] 
         self.technologie=[]

     def construire(self):
        self.tempsConstructionRestant=self.tempsConstructionRestant-1
        self.etatAffichage="(construction) "+str(self.tempsConstructionRestant)
        if self.tempsConstructionRestant == 0:
            #print "CONSTRUCTION TERMINEE"
            self.etatAffichage="OK"
            if type=="extracteurGaz":
                self.etat="RECOLTER"
            elif type=="extracteurMinerai":
                self.etat="RECOLTER"
            elif type=="agricultureFerme":
                self.etat="RECOLTER"
            else:
                self.etat="CONSTRUIT"
                         
     def ajouterTechnologie(self, technologie):
         self.technologie.append(technologie)
        
